import argparse

class Parser(object):
    @staticmethod
    def args():
        parser = argparse.ArgumentParser(description="predict the task.")
        parser.add_argument("-d", "--data", type=str, default="/media/Harddisk_A/goog/Datasets/cat121/cat_12_train" ,help="file folder")
        parser.add_argument("-e", "--epochs", type=int, default=1000 ,help="epochs")
        parser.add_argument("-b", "--batch_size", type=int, default=16 ,help="batch-size")
        parser.add_argument("--save_interval", type=int, default=1 ,help="save interval")
        parser.add_argument("--device", default='0', help="cuda device, i.e. 0 or 0,1,2,3 or cpu")
        parser.add_argument("--seed", type=int, default=0, help="seed")
        parser.add_argument('--log_dir', default='./output_dir',
                        help='path where to tensorboard log')

        # Optimizer parameters
        parser.add_argument('--weight_decay', type=float, default=0.05,
                            help='weight decay (default: 0.05)')
        parser.add_argument('--lr', type=float, default=None, metavar='LR',
                            help='learning rate (absolute lr)')
        args = parser.parse_args()

        return args